package cn.tedu.permutation;

import java.util.*;

/**
 * 给定一个可包含重复数字的序列，返回所有不重复的全排列。
 * <p>
 * 示例:
 * <p>
 * 输入: [A,A,B]
 * 输出:
 * [
 * [A,A,B],
 * [A,B,A],
 * [B,A,A]
 * ]
 */
public class PermutationsIIForChar {
    public List<List<Character>> permuteUnique(char[] arr) {
        int len = arr.length;
        List<List<Character>> res = new ArrayList<>();
        if (len == 0) {
            return res;
        }

        // 排序（升序或者降序都可以），排序是剪枝的前提
        Arrays.sort(arr);

        boolean[] used = new boolean[len];
        // 使用 Deque 是 Java 官方 Stack 类的建议
        Deque<Character> path = new ArrayDeque<>(len);
        dfs(arr, len, 0, used, path, res);
        return res;
    }

    private void dfs(char[] arr, int len, int depth, boolean[] used, Deque<Character> path, List<List<Character>> res) {
        if (depth == len) {
            res.add(new ArrayList<>(path));
            return;
        }

        for (int i = 0; i < len; ++i) {
            if (used[i]) {
                continue;
            }

            // 剪枝条件：i > 0 是为了保证 nums[i - 1] 有意义
            // 写 !used[i - 1] 是因为 nums[i - 1] 在深度优先遍历的过程中刚刚被撤销选择
            if (i > 0 && arr[i] == arr[i - 1] && !used[i - 1]) {
                continue;
            }

            path.push(arr[i]);
            used[i] = true;

            dfs(arr, len, depth + 1, used, path, res);
            // 回溯部分的代码，和 dfs 之前的代码是对称的
            used[i] = false;
            path.pop();
        }
    }

    public static void main(String[] args) {
        PermutationsIIForChar permutationsIIForChar = new PermutationsIIForChar();
        char[] arr = {'A', 'B', 'A'};
        List<List<Character>> res = permutationsIIForChar.permuteUnique(arr);
        System.out.println(res);
    }
}
